import torch

#
dtype = torch.float
device = torch.device("cpu")


def test1():
    data = [[1, 2], [3, 4]]
    x_data = torch.tensor(data)
    print(x_data)


def test2():
    shap = (2, 3,)
    rand_tensor = torch.randn(shap)
    print(rand_tensor)

    ones_tensor = torch.ones(shap)
    print(ones_tensor)

    zeros_tensor = torch.zeros(shap)
    print(zeros_tensor)
